#技术难点
#每次请求12306的验证码图片都会被刷新，导致识别解码的图片不一致
#所以不能单独对该验证码图片进行网络请求，而是用截图的方式

from selenium import webdriver
import requests
from hashlib import md5
import time
from PIL import Image
from selenium.webdriver import ActionChains

class Chaojiying_Client(object):

    def __init__(self, username, password, soft_id):
        self.username = username
        password =  password.encode('utf8')
        self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
        return r.json()




print('打开12306，成功')
bro = webdriver.Chrome(executable_path='./chromedriver.exe')
bro.get('https://kyfw.12306.cn/otn/resources/login.html')


login_btn = bro.find_element_by_xpath('/html/body/div[2]/div[2]/ul/li[2]/a')
login_btn.click()
time.sleep(0.5)
#save_screenshot将当前页面截图且保存

bro.maximize_window()
print('最大化窗口')
time.sleep(0.5)
bro.save_screenshot('aa.png')
print('保存截屏')
#确定验证码图片对应的左上角和右下角坐标（裁剪的区域就确定）
code_img_ele = bro.find_element_by_xpath('//*[@id="J-loginImg"]')
location = code_img_ele.location  #验证码图片左上角的图标x，y
size = code_img_ele.size   #验证码对应的长和宽
print('获取验证码图片')
rangle = (int(location['x']),int(location['y']),int(location['x']+size['width']),int(location['y']+size['height']))
#至此验证码图片确定下来了

#实例化一个image对象
i = Image.open('./aa.png')
code_img_name = './code.png'
#根据指定区域进行图片裁剪
frame = i.crop(rangle)
frame.save(code_img_name)

print('启动云解析验证码平台进行解码...')
#验证码给超级鹰识别
chaojiying = Chaojiying_Client('flatts', 'war321', '914461')  # 用户中心>>软件ID 生成一个替换 96001
im = open('code.png', 'rb').read()  # 本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
print(chaojiying.PostPic(im, 9004)['pic_str'])  #
result = chaojiying.PostPic(im, 9004)['pic_str']
print('验证码解析成功')

all_list = []#存储要被点击的点的坐标[[x1,y1],[x2,y2]}

if '|' in result:
    list_1 = result.split('|')
    count_1 = len(list_1)
    for i in range(count_1):
        xy_list = []
        x = int(list_1[i].split(',')[0])
        y = int(list_1[i].split(',')[0])
        xy_list.append(x)
        xy_list.append(y)
        all_list.append(xy_list)
else:
    x = int(result.split(',')[0])
    y = int(result.split(',')[1])
    xy_list = []
    xy_list.append(x)
    xy_list.append(y)
    all_list.append(xy_list)


#遍历列表，使用动作链对每个列表元素的xy指定位置进行操作
for l in all_list:
    x = l[0]
    y = l[1]
    ActionChains(bro).move_to_element_with_offset(code_img_ele,x,y).click().perform()
    time.sleep(0.5)

bro.find_element_by_id('J-userName').send_keys('Aliam')
time.sleep(1)
bro.find_element_by_id('J-password').send_keys('howareyou')
time.sleep(1)
bro.find_element_by_id('J-login').click()
time.sleep(1)
